// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <stddef.h>

#include "cc/layers/append_quads_data.h"
#include "cc/layers/heads_up_display_layer_impl.h"
#include "cc/test/fake_compositor_frame_sink.h"
#include "cc/test/fake_impl_task_runner_provider.h"
#include "cc/test/fake_layer_tree_host_impl.h"
#include "cc/test/test_task_graph_runner.h"
#include "cc/trees/layer_tree_impl.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace cc {
namespace {

    void CheckDrawLayer(HeadsUpDisplayLayerImpl* layer,
        ResourceProvider* resource_provider,
        DrawMode draw_mode)
    {
        std::unique_ptr<RenderPass> render_pass = RenderPass::Create();
        AppendQuadsData data;
        bool will_draw = layer->WillDraw(draw_mode, resource_provider);
        if (will_draw)
            layer->AppendQuads(render_pass.get(), &data);
        layer->UpdateHudTexture(draw_mode, resource_provider);
        if (will_draw)
            layer->DidDraw(resource_provider);

        size_t expected_quad_list_size = will_draw ? 1 : 0;
        EXPECT_EQ(expected_quad_list_size, render_pass->quad_list.size());
    }

    TEST(HeadsUpDisplayLayerImplTest, ResourcelessSoftwareDrawAfterResourceLoss)
    {
        FakeImplTaskRunnerProvider task_runner_provider;
        TestTaskGraphRunner task_graph_runner;
        std::unique_ptr<CompositorFrameSink> compositor_frame_sink = FakeCompositorFrameSink::Create3d();
        FakeLayerTreeHostImpl host_impl(&task_runner_provider, &task_graph_runner);
        host_impl.CreatePendingTree();
        host_impl.SetVisible(true);
        host_impl.InitializeRenderer(compositor_frame_sink.get());
        std::unique_ptr<HeadsUpDisplayLayerImpl> layer_ptr = HeadsUpDisplayLayerImpl::Create(host_impl.pending_tree(), 1);
        layer_ptr->SetBounds(gfx::Size(100, 100));

        HeadsUpDisplayLayerImpl* layer = layer_ptr.get();

        host_impl.pending_tree()->SetRootLayerForTesting(std::move(layer_ptr));
        host_impl.pending_tree()->BuildLayerListAndPropertyTreesForTesting();

        // Check regular hardware draw is ok.
        CheckDrawLayer(layer, host_impl.resource_provider(), DRAW_MODE_HARDWARE);

        // Simulate a resource loss on transitioning to resourceless software mode.
        layer->ReleaseResources();

        // Should skip resourceless software draw and not crash in UpdateHudTexture.
        CheckDrawLayer(layer, host_impl.resource_provider(),
            DRAW_MODE_RESOURCELESS_SOFTWARE);
    }

} // namespace
} // namespace cc
